
import paddle
from paddlenlp.embeddings import TokenEmbedding, list_embedding_name
from scipy import spatial

paddle.set_device("cpu")
print(list_embedding_name())

tar_str = "飞桨以百度多年的深度学习技术研究和业务应用为基础，集核心框架、基础模型库、端到端开发套件、丰富的工具组件、星河社区于一体，是中国首个自主研发、功能丰富、开源开放的产业级深度学习平台"
tar_str2 = "飞桨以百度多年的深度学习技术研究,星河社区于一体"
tar_str3 = "作为老的四星酒店，房间依然很整洁，相当不错。机场接机服务很好，可以在车上办理入住手续，节省时间。"

token_embedding = TokenEmbedding(embedding_name="w2v.baidu_encyclopedia.target.word-word.dim300")
test_token_embedding = token_embedding.search(tar_str)
print(test_token_embedding)

score = token_embedding.cosine_sim("中国", "美国")
print(score)


#labels = token_embedding.vocab.to_tokens(list(range(0,1000)))
#from visualdl import LogWriter
#with LogWriter(logdir='./visualize') as writer:
#    writer.add_embeddings(tag='test', mat=test_token_embedding, metadata=labels)